tensor([[0.4500, 0.4500, 0.4500, 0.4500, 0.4500],
        [0.5500, 0.5500, 0.5500, 0.5500, 0.5500]], grad_fn=<RoundBackward1>)